import json

from tool import get_pre_prompt, get_replace, get_questions, replace_func


'''
生成一次性问18个问题的prompt, deepseek用中文, gpt用英文
'''


if __name__ == "__main__":
    pre_prompt_zh = get_pre_prompt("pre_prompt_no_country_zh")
    pre_prompt_en = get_pre_prompt("pre_prompt_no_country_en")
    questions_zh = get_questions()
    questions_en = get_questions("enquestions")
    prompts = {
        "deepseek-reasoner": pre_prompt_zh + "\n" + "\n".join(questions_zh),
        "gpt-4o": pre_prompt_en + "\n" + "\n".join(questions_en)}
    # 写入文件
    with open("tmp/prompts_v3.json", "w", encoding="utf8") as f:
        json.dump(prompts, f, ensure_ascii=False)
